change some info store in redis & remove some unnecessary params check & add some return for some api

Brightcells 9 anos atrás
pai
commit
454f8cd8ec

+ 2 - 3
account/models.py

@@ -138,15 +138,14 @@ class UserInfo(CreateUpdateMixin):
138 138
         elif self.user_from == self.WX_USER:
139 139
             return self.nickname
140 140
 
141
-    def _data(self):
141
+    @property
142
+    def data(self):
142 143
         return {
143 144
             'user_id': self.user_id,
144 145
             'username': self.username,
145 146
             'nickname': self.nickname,
146 147
         }
147 148
 
148
-    data = property(_data)
149
-
150 149
 
151 150
 class UserLoginLogInfo(CreateUpdateMixin):
152 151
     SUCCESS = 0

+ 1 - 1
account/views.py

@@ -11,7 +11,7 @@ from account.serializers import UserSerializer, GroupSerializer, LensmanInfoSeri
11 11
 
12 12
 from utils.ip_utils import ip_addr
13 13
 
14
-from utils.error.errno_utils import LensmanStatusCode
14
+from utils.error.errno_utils import LensmanStatusCode, UserStatusCode
15 15
 from utils.error.response_utils import response
16 16
 
17 17
 from curtail_uuid import CurtailUUID

+ 2 - 2
api/urls.py

@@ -30,8 +30,8 @@ urlpatterns += [
30 30
     url(r'^g/lock$', group_views.group_lock_api, name='group_lock_api'),  # 群组锁定
31 31
     url(r'^g/unlock$', group_views.group_unlock_api, name='group_unlock_api'),  # 群组解锁
32 32
     url(r'^g/remove$', group_views.group_remove_api, name='group_remove_api'),  # 成员移除
33
-    url(r'^g/pass$', group_views.group_pass_api, name='group_pass_api'),  # 申请通过
34
-    url(r'^g/refuse$', group_views.group_refuse_api, name='group_refuse_api'),  # 申请拒绝
33
+    # url(r'^g/pass$', group_views.group_pass_api, name='group_pass_api'),  # 申请通过
34
+    # url(r'^g/refuse$', group_views.group_refuse_api, name='group_refuse_api'),  # 申请拒绝
35 35
 ]
36 36
 
37 37
 # 飞图相关

+ 3 - 2
group/models.py

@@ -43,6 +43,7 @@ class GroupInfo(CreateUpdateMixin):
43 43
     def data(self):
44 44
         return {
45 45
             'group_id': self.group_id,
46
+            'admin_id': self.admin_id,
46 47
             'group_name': self.group_name,
47 48
             'group_default_avatar': self.group_default_avatar,
48 49
             'group_avatar': self.group_avatar_url,
@@ -52,14 +53,14 @@ class GroupInfo(CreateUpdateMixin):
52 53
             'created_at': self.created_at,
53 54
         }
54 55
 
55
-    def users(self, user_id):
56
+    def users(self, admin=True, user_id=None):
56 57
         all_users = GroupUserInfo.objects.filter(group_id=self.group_id)
57 58
 
58 59
         passed_users = all_users.filter(user_status=GroupUserInfo.PASSED)
59 60
         passed_count = passed_users.count()
60 61
         passed = [passed.user_info for passed in passed_users]
61 62
 
62
-        if self.admin_id != user_id:
63
+        if admin and self.admin_id != user_id:
63 64
             return {
64 65
                 'passed_count': passed_count,
65 66
                 'passed': passed,

+ 276 - 174
group/views.py

@@ -20,6 +20,10 @@ from utils.url_utils import img_url
20 20
 from utils.error.errno_utils import UserStatusCode, GroupStatusCode, GroupUserStatusCode, GroupPhotoStatusCode
21 21
 from utils.error.response_utils import response
22 22
 
23
+from utils.redis.rkeys import GROUP_USERS_APPLYING_SET, GROUP_USERS_PASSED_SET, GROUP_USERS_REFUSED_SET, GROUP_USERS_DELETED_SET
24
+from utils.redis.rkeys import GROUP_LAST_PHOTO_PK
25
+from utils.redis.rgroup import set_group_info, get_group_info, set_group_users_info, get_group_users_info
26
+
23 27
 from utils.sql.raw import PAI2_HOME_API
24 28
 
25 29
 from curtail_uuid import CurtailUUID
@@ -35,16 +39,25 @@ r = settings.REDIS_CACHE
35 39
 
36 40
 @transaction.atomic
37 41
 def group_create_api(request):
42
+    """
43
+    群组创建
44
+    :param request:
45
+    :return:
46
+    """
38 47
     user_id = request.POST.get('user_id', '')
39 48
     group_name = request.POST.get('group_name', '')
40 49
     group_default_avatar = int(request.POST.get('group_default_avatar', 0))
41 50
 
51
+    # 用户校验
42 52
     try:
43 53
         user = UserInfo.objects.get(user_id=user_id)
44 54
     except UserInfo.DoesNotExist:
45 55
         return response(UserStatusCode.USER_NOT_FOUND)
46 56
 
57
+    # 群组唯一标识
47 58
     group_id = CurtailUUID.uuid(GroupInfo, 'group_id')
59
+
60
+    # 群组记录创建
48 61
     group = GroupInfo.objects.create(
49 62
         group_id=group_id,
50 63
         admin_id=user_id,
@@ -52,6 +65,11 @@ def group_create_api(request):
52 65
         group_default_avatar=group_default_avatar,
53 66
         group_from=GroupInfo.APP_GROUP,
54 67
     )
68
+
69
+    # Redis 群组数据缓存
70
+    group_info = set_group_info(group)
71
+
72
+    # 群组用户记录创建
55 73
     GroupUserInfo.objects.create(
56 74
         group_id=group_id,
57 75
         user_id=user_id,
@@ -62,50 +80,46 @@ def group_create_api(request):
62 80
         passed_at=tc.utc_datetime(),
63 81
     )
64 82
 
83
+    # Redis 群组用户数据缓存
84
+    group_users = set_group_users_info(group)
85
+
65 86
     return JsonResponse({
66 87
         'status': 200,
67 88
         'message': u'群组创建成功',
68 89
         'data': {
69 90
             'group_id': group_id,
70
-            'group': group.data,
71
-            'users': group.users(user_id),
91
+            'group': group_info,
92
+            'users': group_users,
72 93
         },
73 94
     })
74 95
 
75 96
 
76 97
 def group_detail_api(request):
98
+    """
99
+    群组详情
100
+    :param request:
101
+    :return:
102
+    """
77 103
     group_id = request.POST.get('group_id', '')
78 104
     user_id = request.POST.get('user_id', '')
79 105
 
80
-    try:
81
-        user = UserInfo.objects.get(user_id=user_id)
82
-    except UserInfo.DoesNotExist:
83
-        return response(UserStatusCode.USER_NOT_FOUND)
84
-
85
-    try:
86
-        group = GroupInfo.objects.get(group_id=group_id)
87
-    except GroupInfo.DoesNotExist:
88
-        return response(GroupStatusCode.GROUP_NOT_FOUND)
89
-
90
-    # try:
91
-    #     group_user = GroupUserInfo.objects.filter(group_id=group_id, user_id=user_id)
92
-    # except GroupUserInfo.DoesNotExist:
93
-    #     return response(GroupUserStatusCode.GROUP_USER_NOT_FOUND)
94
-    # if group_user.user_status == GroupUserInfo.DELETED:
95
-    #     return response(GroupUserStatusCode.GROUP_USER_HAS_DELETED)
96
-
97 106
     return JsonResponse({
98 107
         'status': 200,
99 108
         'message': u'获取群组详情成功',
100 109
         'data': {
101 110
             'group_id': group_id,
102
-            'group': group.data,
103
-            'users': group.users(user_id),
111
+            'group': get_group_info(group_id),
112
+            'users': get_group_users_info(group_id, user_id),
104 113
         },
105 114
     })
106 115
 
107 116
 
108 117
 def group_update_api(request):
118
+    """
119
+    群组更新
120
+    :param request:
121
+    :return:
122
+    """
109 123
     group_id = request.POST.get('group_id', '')
110 124
     admin_id = request.POST.get('admin_id', '') or request.POST.get('user_id', '')
111 125
     group_name = request.POST.get('group_name', '')
@@ -113,21 +127,23 @@ def group_update_api(request):
113 127
 
114 128
     group_avatar = request.FILES.get('group_avatar', '')
115 129
 
116
-    try:
117
-        admin = UserInfo.objects.get(user_id=admin_id)
118
-    except UserInfo.DoesNotExist:
119
-        return response(UserStatusCode.USER_NOT_FOUND)
120
-
130
+    # 群组校验
121 131
     try:
122 132
         group = GroupInfo.objects.get(group_id=group_id)
123 133
     except GroupInfo.DoesNotExist:
124 134
         return response(GroupStatusCode.GROUP_NOT_FOUND)
135
+
136
+    # 权限校验
125 137
     if group.admin_id != admin_id:
126 138
         return response(GroupStatusCode.NO_UPDATE_PERMISSION)
139
+
140
+    # 群组名称更新
127 141
     if group_name:
128 142
         group.group_name = group_name
143
+    # 群组描述更新
129 144
     if group_desc:
130 145
         group.group_desc = group_desc
146
+    # 群组头像更新
131 147
     if group_avatar:
132 148
         _, extension = os.path.splitext(group_avatar.name)
133 149
         group_avatar_path = 'group/{uuid}_{extension}'.format(uuid=shortuuid.uuid(), extension=extension)
@@ -137,27 +153,33 @@ def group_update_api(request):
137 153
         group.group_avatar = group_avatar_path
138 154
     group.save()
139 155
 
156
+    # Redis 群组数据缓存更新
157
+    group_info = set_group_info(group)
158
+
140 159
     return JsonResponse({
141 160
         'status': 200,
142
-        'message': u'用户更新成功',
161
+        'message': u'群组更新成功',
143 162
         'data': {
144 163
             'group_id': group_id,
145
-            'group': group.data,
146
-            'users': group.users(admin_id),
164
+            'group': group_info,
165
+            'users': get_group_users_info(group_id, admin_id),
147 166
         },
148 167
     })
149 168
 
150 169
 
151 170
 def group_list_api(request):
171
+    """
172
+    群组列表
173
+    :param request:
174
+    :return:
175
+    """
152 176
     user_id = request.POST.get('user_id', '')
153 177
 
154 178
     group_users = GroupUserInfo.objects.filter(user_id=user_id, user_status=GroupUserInfo.PASSED)
155 179
     groups = []
156 180
     for group_user in group_users:
157
-        try:
158
-            groups.append(GroupInfo.objects.get(group_id=group_user.group_id).data)
159
-        except GroupInfo.DoesNotExist:
160
-            pass
181
+        group_info = get_group_info(group_user.group_id)
182
+        groups.append(group_info) if group_info else None
161 183
 
162 184
     return JsonResponse({
163 185
         'status': 200,
@@ -169,31 +191,40 @@ def group_list_api(request):
169 191
 
170 192
 
171 193
 def group_join_api(request):
194
+    """
195
+    申请加群
196
+    :param request:
197
+    :return:
198
+    """
172 199
     group_id = request.POST.get('group_id', '')
173 200
     user_id = request.POST.get('user_id', '')
174 201
     nickname = request.POST.get('nickname', '')
175 202
 
203
+    # 用户校验
176 204
     try:
177 205
         user = UserInfo.objects.get(user_id=user_id)
178 206
     except UserInfo.DoesNotExist:
179 207
         return response(UserStatusCode.USER_NOT_FOUND)
180 208
 
209
+    # 群组校验
181 210
     try:
182 211
         group = GroupInfo.objects.get(group_id=group_id)
183 212
     except GroupInfo.DoesNotExist:
184 213
         return response(GroupStatusCode.GROUP_NOT_FOUND)
214
+
215
+    # 群组锁定校验
185 216
     if group.group_lock:
186 217
         return response(GroupStatusCode.GROUP_HAS_LOCKED)
187 218
 
188
-    if GroupUserInfo.objects.filter(group_id=group_id, user_id=user_id, user_status__in=[GroupUserInfo.APPLYING, GroupUserInfo.PASSED]).exists():
219
+    # 重复申请校验
220
+    if r.sismember(GROUP_USERS_APPLYING_SET % group_id, user_id) or r.sismember(GROUP_USERS_PASSED_SET % group_id, user_id):
189 221
         return response(GroupStatusCode.DUPLICATE_JOIN_REQUEST)
190 222
 
191
-    group_photo = GroupPhotoInfo.objects.filter(group_id=group_id).last()
192
-
223
+    # 群组用户记录创建
193 224
     GroupUserInfo.objects.create(
194 225
         group_id=group_id,
195 226
         user_id=user_id,
196
-        current_id=group_photo and group_photo.id or -1,
227
+        current_id=int(r.get(GROUP_LAST_PHOTO_PK % group_id) or -1),
197 228
         nickname=nickname or user.final_nickname,
198 229
         avatar=user.avatar,
199 230
         admin=False,
@@ -201,34 +232,49 @@ def group_join_api(request):
201 232
         passed_at=tc.utc_datetime(),
202 233
     )
203 234
 
235
+    # Redis 群组用户数据缓存
236
+    set_group_users_info(group)
237
+
238
+    # Redis 群组通过集合缓存
239
+    r.sadd(GROUP_USERS_PASSED_SET % group_id, user_id)
240
+
204 241
     return JsonResponse({
205 242
         'status': 200,
206
-        'message': u'申请已提交',
243
+        'message': u'申请成功',
244
+        'data': {
245
+            'group_id': group_id,
246
+            'group': get_group_info(group_id),
247
+            'users': get_group_users_info(group_id, user_id),
248
+        },
207 249
     })
208 250
 
209 251
 
210 252
 def group_lock_api(request):
253
+    """
254
+    群组锁定
255
+    :param request:
256
+    :return:
257
+    """
211 258
     group_id = request.POST.get('group_id', '')
212
-    user_id = request.POST.get('user_id', '')
213
-
214
-    try:
215
-        user = UserInfo.objects.get(user_id=user_id)
216
-    except UserInfo.DoesNotExist:
217
-        return response(UserStatusCode.USER_NOT_FOUND)
259
+    admin_id = request.POST.get('admin_id', '') or request.POST.get('user_id', '')
218 260
 
261
+    # 群组校验
219 262
     try:
220 263
         group = GroupInfo.objects.get(group_id=group_id)
221 264
     except GroupInfo.DoesNotExist:
222 265
         return response(GroupStatusCode.GROUP_NOT_FOUND)
223
-    if group.admin_id != user_id:
224
-        return JsonResponse({
225
-            'status': 40221,
226
-            'message': u'没有锁定权限',
227
-        })
228 266
 
267
+    # 权限校验
268
+    if group.admin_id != admin_id:
269
+        return response(GroupStatusCode.NO_LOCK_PERMISSION)
270
+
271
+    # 群组锁定
229 272
     group.group_lock = True
230 273
     group.save()
231 274
 
275
+    # Redis 群组数据缓存更新
276
+    set_group_info(group)
277
+
232 278
     return JsonResponse({
233 279
         'status': 200,
234 280
         'message': u'锁定成功',
@@ -236,24 +282,31 @@ def group_lock_api(request):
236 282
 
237 283
 
238 284
 def group_unlock_api(request):
285
+    """
286
+    群组解锁
287
+    :param request:
288
+    :return:
289
+    """
239 290
     group_id = request.POST.get('group_id', '')
240
-    user_id = request.POST.get('user_id', '')
241
-
242
-    try:
243
-        user = UserInfo.objects.get(user_id=user_id)
244
-    except UserInfo.DoesNotExist:
245
-        return response(UserStatusCode.USER_NOT_FOUND)
291
+    admin_id = request.POST.get('admin_id', '') or request.POST.get('user_id', '')
246 292
 
293
+    # 群组校验
247 294
     try:
248 295
         group = GroupInfo.objects.get(group_id=group_id)
249 296
     except GroupInfo.DoesNotExist:
250 297
         return response(GroupStatusCode.GROUP_NOT_FOUND)
251
-    if group.admin_id != user_id:
298
+
299
+    # 权限校验
300
+    if group.admin_id != admin_id:
252 301
         return response(GroupStatusCode.NO_UNLOCK_PERMISSION)
253 302
 
303
+    # 群组解锁
254 304
     group.group_lock = False
255 305
     group.save()
256 306
 
307
+    # Redis 群组数据缓存更新
308
+    set_group_info(group)
309
+
257 310
     return JsonResponse({
258 311
         'status': 200,
259 312
         'message': u'解锁成功',
@@ -261,126 +314,155 @@ def group_unlock_api(request):
261 314
 
262 315
 
263 316
 def group_remove_api(request):
317
+    """
318
+    成员移除
319
+    :param request:
320
+    :return:
321
+    """
264 322
     group_id = request.POST.get('group_id', '')
265 323
     admin_id = request.POST.get('admin_id', '')
266 324
     user_id = request.POST.get('user_id', '')
267 325
 
268
-    try:
269
-        admin = UserInfo.objects.get(user_id=admin_id)
270
-    except UserInfo.DoesNotExist:
271
-        return response(UserStatusCode.USER_NOT_FOUND)
272
-    try:
273
-        user = UserInfo.objects.get(user_id=user_id)
274
-    except UserInfo.DoesNotExist:
275
-        return response(UserStatusCode.USER_NOT_FOUND)
276
-
326
+    # 群组校验
277 327
     try:
278 328
         group = GroupInfo.objects.get(group_id=group_id)
279 329
     except GroupInfo.DoesNotExist:
280 330
         return response(GroupStatusCode.GROUP_NOT_FOUND)
331
+
332
+    # 权限校验
281 333
     if group.admin_id != admin_id:
282
-        return JsonResponse({
283
-            'status': 40222,
284
-            'message': u'没有移除权限',
285
-        })
334
+        return response(GroupStatusCode.NO_REMOVE_PERMISSION)
286 335
 
336
+    # 群组用户校验
287 337
     try:
288 338
         group_user = GroupUserInfo.objects.get(group_id=group_id, user_id=user_id, user_status=GroupUserInfo.PASSED)
289 339
     except GroupUserInfo.DoesNotExist:
290
-        return response(GroupStatusCode.GROUP_USER_NOT_FOUND)
340
+        return response(GroupUserStatusCode.GROUP_USER_NOT_FOUND)
341
+
342
+    # 群组用户移除
291 343
     group_user.user_status = GroupUserInfo.DELETED
292 344
     group_user.deleted_at = tc.utc_datetime()
293 345
     group_user.save()
294 346
 
347
+    # Redis 群组数据缓存更新
348
+    group_users = set_group_info(group)
349
+
350
+    # Redis 群组删除集合缓存
351
+    r.srem(GROUP_USERS_PASSED_SET % group_id, user_id)
352
+    r.sadd(GROUP_USERS_DELETED_SET % group_id, user_id)
353
+
295 354
     return JsonResponse({
296 355
         'status': 200,
297 356
         'message': u'用户移除成功',
298 357
         'data': {
299 358
             'group_id': group_id,
300
-            'users': group.users(admin_id),
359
+            'users': group_users,
301 360
         },
302 361
     })
303 362
 
304 363
 
305 364
 def group_pass_api(request):
365
+    """
366
+    申请通过
367
+    :param request:
368
+    :return:
369
+    """
306 370
     group_id = request.POST.get('group_id', '')
307 371
     admin_id = request.POST.get('admin_id', '')
308 372
     user_id = request.POST.get('user_id', '')
309 373
 
310
-    try:
311
-        admin = UserInfo.objects.get(user_id=admin_id)
312
-    except UserInfo.DoesNotExist:
313
-        return response(UserStatusCode.USER_NOT_FOUND)
314
-    try:
315
-        user = UserInfo.objects.get(user_id=user_id)
316
-    except UserInfo.DoesNotExist:
317
-        return response(UserStatusCode.USER_NOT_FOUND)
318
-
374
+    # 群组校验
319 375
     try:
320 376
         group = GroupInfo.objects.get(group_id=group_id)
321 377
     except GroupInfo.DoesNotExist:
322 378
         return response(GroupStatusCode.GROUP_NOT_FOUND)
379
+
380
+    # 权限校验
323 381
     if group.admin_id != admin_id:
324 382
         return response(GroupStatusCode.NO_PASS_PERMISSION)
325 383
 
384
+    # 群组用户校验
326 385
     try:
327 386
         group_user = GroupUserInfo.objects.get(group_id=group_id, user_id=user_id, user_status=GroupUserInfo.APPLYING)
328 387
     except GroupUserInfo.DoesNotExist:
329 388
         return response(GroupStatusCode.JOIN_REQUEST_NOT_FOUND)
389
+
390
+    # 群组用户通过
330 391
     group_user.user_status = GroupUserInfo.PASSED
331 392
     group_user.passed_at = tc.utc_datetime()
332 393
     group_user.save()
333 394
 
395
+    # Redis 群组数据缓存更新
396
+    group_users = set_group_info(group)
397
+
398
+    # Redis 群组通过集合缓存
399
+    r.srem(GROUP_USERS_APPLYING_SET % group_id, user_id)
400
+    r.sadd(GROUP_USERS_PASSED_SET % group_id, user_id)
401
+
334 402
     return JsonResponse({
335 403
         'status': 200,
336 404
         'message': u'申请通过成功',
337 405
         'data': {
338 406
             'group_id': group_id,
339
-            'users': group.users(admin_id),
407
+            'users': group_users,
340 408
         },
341 409
     })
342 410
 
343 411
 
344 412
 def group_refuse_api(request):
413
+    """
414
+    申请拒绝
415
+    :param request:
416
+    :return:
417
+    """
345 418
     group_id = request.POST.get('group_id', '')
346 419
     admin_id = request.POST.get('admin_id', '')
347 420
     user_id = request.POST.get('user_id', '')
348 421
 
349
-    try:
350
-        admin = UserInfo.objects.get(user_id=admin_id)
351
-    except UserInfo.DoesNotExist:
352
-        return response(UserStatusCode.USER_NOT_FOUND)
353
-    try:
354
-        user = UserInfo.objects.get(user_id=user_id)
355
-    except UserInfo.DoesNotExist:
356
-        return response(UserStatusCode.USER_NOT_FOUND)
357
-
422
+    # 群组校验
358 423
     try:
359 424
         group = GroupInfo.objects.get(group_id=group_id)
360 425
     except GroupInfo.DoesNotExist:
361 426
         return response(GroupStatusCode.GROUP_NOT_FOUND)
427
+
428
+    # 权限校验
362 429
     if group.admin_id != admin_id:
363 430
         return response(GroupStatusCode.NO_REFUSE_PERMISSION)
364 431
 
432
+    # 群组用户校验
365 433
     try:
366 434
         group_user = GroupUserInfo.objects.get(group_id=group_id, user_id=user_id, user_status=GroupUserInfo.APPLYING)
367 435
     except GroupUserInfo.DoesNotExist:
368 436
         return response(GroupStatusCode.JOIN_REQUEST_NOT_FOUND)
437
+
438
+    # 群组用户拒绝
369 439
     group_user.user_status = GroupUserInfo.REFUSED
370 440
     group_user.refused_at = tc.utc_datetime()
371 441
     group_user.save()
372 442
 
443
+    # Redis 群组数据缓存更新
444
+    group_users = set_group_info(group)
445
+
446
+    # Redis 群组拒绝集合缓存
447
+    r.srem(GROUP_USERS_APPLYING_SET % group_id, user_id)
448
+    r.sadd(GROUP_USERS_REFUSED_SET % group_id, user_id)
449
+
373 450
     return JsonResponse({
374 451
         'status': 200,
375 452
         'message': u'申请拒绝成功',
376 453
         'data': {
377 454
             'group_id': group_id,
378
-            'users': group.users(admin_id),
455
+            'users': group_users,
379 456
         },
380 457
     })
381 458
 
382 459
 
383 460
 def flyimg_upload_api(request):
461
+    """
462
+    飞图上传/飞图列表
463
+    :param request:
464
+    :return:
465
+    """
384 466
     group_id = request.POST.get('group_id', '')
385 467
     user_id = request.POST.get('user_id', '')
386 468
     nickname = request.POST.get('nickname', '')
@@ -389,20 +471,17 @@ def flyimg_upload_api(request):
389 471
 
390 472
     current_id = int(request.POST.get('current_id', -1))
391 473
 
474
+    # 用户校验
392 475
     try:
393 476
         user = UserInfo.objects.get(user_id=user_id)
394 477
     except UserInfo.DoesNotExist:
395 478
         return response(UserStatusCode.USER_NOT_FOUND)
396 479
 
397
-    try:
398
-        group = GroupInfo.objects.get(group_id=group_id)
399
-    except GroupInfo.DoesNotExist:
400
-        return response(GroupStatusCode.GROUP_NOT_FOUND)
401
-
480
+    # 群组用户校验
402 481
     try:
403 482
         group_user = GroupUserInfo.objects.get(group_id=group_id, user_id=user_id, user_status=GroupUserInfo.PASSED)
404 483
     except GroupUserInfo.DoesNotExist:
405
-        return response(GroupStatusCode.GROUP_USER_NOT_FOUND)
484
+        return response(GroupUserStatusCode.GROUP_USER_NOT_FOUND)
406 485
 
407 486
     if photo:
408 487
         _, extension = os.path.splitext(photo.name)
@@ -424,6 +503,7 @@ def flyimg_upload_api(request):
424 503
             settings.THUMBNAIL_MAX_WIDTH
425 504
         )
426 505
 
506
+        # 群组照片记录创建
427 507
         group_photo = GroupPhotoInfo.objects.create(
428 508
             group_id=group_id,
429 509
             user_id=user_id,
@@ -437,6 +517,10 @@ def flyimg_upload_api(request):
437 517
             photo_thumbnail_h=photo_thumbnail_h,
438 518
         )
439 519
 
520
+        # 设置群组最后一张照片PK
521
+        r.set(GROUP_LAST_PHOTO_PK % group_id, group_photo.pk)
522
+
523
+    # 获取从 current_id 到 now 的群组照片列表
440 524
     group_photos = GroupPhotoInfo.objects.filter(
441 525
         group_id=group_id,
442 526
         status=True,
@@ -455,27 +539,30 @@ def flyimg_upload_api(request):
455 539
 
456 540
 
457 541
 def comment_submit_api(request):
542
+    """
543
+    飞图评论提交/飞图评论列表
544
+    :param request:
545
+    :return:
546
+    """
458 547
     group_id = request.POST.get('group_id', '')
459 548
     user_id = request.POST.get('user_id', '')
460 549
     photo_id = request.POST.get('photo_id', '')
461 550
     comment = request.POST.get('comment', '')
462 551
 
463
-    try:
464
-        group = GroupInfo.objects.get(group_id=group_id)
465
-    except GroupInfo.DoesNotExist:
466
-        return response(GroupStatusCode.GROUP_NOT_FOUND)
467
-
552
+    # 群组用户校验
468 553
     try:
469 554
         group_user = GroupUserInfo.objects.get(group_id=group_id, user_id=user_id, user_status=GroupUserInfo.PASSED)
470 555
     except GroupUserInfo.DoesNotExist:
471
-        return response(GroupStatusCode.GROUP_USER_NOT_FOUND)
556
+        return response(GroupUserStatusCode.GROUP_USER_NOT_FOUND)
472 557
 
558
+    # 群组照片校验
473 559
     try:
474 560
         group_photo = GroupPhotoInfo.objects.get(pk=photo_id)
475 561
     except GroupPhotoInfo.DoesNotExist:
476
-        return response(GroupPhotoStatusCode.MESSAGE_NOT_FOUND)
562
+        return response(GroupPhotoStatusCode.GROUP_PHOTO_NOT_FOUND)
477 563
 
478 564
     if comment:
565
+        # 群组照片评论记录创建
479 566
         PhotoCommentInfo.objects.create(
480 567
             photo_id=photo_id,
481 568
             user_id=user_id,
@@ -484,21 +571,25 @@ def comment_submit_api(request):
484 571
             comment=comment,
485 572
         )
486 573
 
574
+        # 群组照片评论数更新
487 575
         group_photo.comment_num += 1
488 576
         group_photo.save()
489 577
 
490
-        UserMessageInfo.objects.create(
491
-            from_uid=user_id,
492
-            from_nickname=group_user.nickname,
493
-            from_avatar=group_user.avatar,
494
-            to_uid=group_photo.user_id,
495
-            group_id=group_photo.group_id,
496
-            photo_id=group_photo.pk,
497
-            msg_type=UserMessageInfo.COMMENT,
498
-            msg_title=u'评论',
499
-            msg_content=comment,
500
-        )
501
-
578
+        # 判断群组照片发布者是否已经被管理员移除,如若移除,则不给发布者提醒
579
+        if r.sismember(GROUP_USERS_PASSED_SET % group_photo.group_id, group_photo.user_id):
580
+            UserMessageInfo.objects.create(
581
+                from_uid=user_id,
582
+                from_nickname=group_user.nickname,
583
+                from_avatar=group_user.avatar,
584
+                to_uid=group_photo.user_id,
585
+                group_id=group_photo.group_id,
586
+                photo_id=group_photo.pk,
587
+                msg_type=UserMessageInfo.COMMENT,
588
+                msg_title=u'评论',
589
+                msg_content=comment,
590
+            )
591
+
592
+    # 群组照片评论列表
502 593
     photo_comments = PhotoCommentInfo.objects.filter(
503 594
         photo_id=photo_id,
504 595
     )
@@ -513,25 +604,28 @@ def comment_submit_api(request):
513 604
 
514 605
 
515 606
 def thumbup_submit_api(request):
607
+    """
608
+    飞图点赞提交
609
+    :param request:
610
+    :return:
611
+    """
516 612
     group_id = request.POST.get('group_id', '')
517 613
     user_id = request.POST.get('user_id', '')
518 614
     photo_id = request.POST.get('photo_id', '')
519 615
 
520
-    try:
521
-        group = GroupInfo.objects.get(group_id=group_id)
522
-    except GroupInfo.DoesNotExist:
523
-        return response(GroupStatusCode.GROUP_NOT_FOUND)
524
-
616
+    # 群组用户校验
525 617
     try:
526 618
         group_user = GroupUserInfo.objects.get(group_id=group_id, user_id=user_id, user_status=GroupUserInfo.PASSED)
527 619
     except GroupUserInfo.DoesNotExist:
528
-        return response(GroupStatusCode.GROUP_USER_NOT_FOUND)
620
+        return response(GroupUserStatusCode.GROUP_USER_NOT_FOUND)
529 621
 
622
+    # 群组照片校验
530 623
     try:
531 624
         group_photo = GroupPhotoInfo.objects.get(pk=photo_id)
532 625
     except GroupPhotoInfo.DoesNotExist:
533
-        return response(GroupPhotoStatusCode.MESSAGE_NOT_FOUND)
626
+        return response(GroupPhotoStatusCode.GROUP_PHOTO_NOT_FOUND)
534 627
 
628
+    # 群组照片点赞记录创建/更新
535 629
     photo_thumbup, created = PhotoThumbUpInfo.objects.get_or_create(
536 630
         photo_id=photo_id,
537 631
         user_id=user_id,
@@ -541,21 +635,25 @@ def thumbup_submit_api(request):
541 635
     photo_thumbup.thumbup = True
542 636
     photo_thumbup.save()
543 637
 
638
+    # 群组照片点赞数更新
544 639
     group_photo.thumbup_num += 1
545 640
     group_photo.save()
546 641
 
547
-    UserMessageInfo.objects.create(
548
-        from_uid=user_id,
549
-        from_nickname=group_user.nickname,
550
-        from_avatar=group_user.avatar,
551
-        to_uid=group_photo.user_id,
552
-        group_id=group_photo.group_id,
553
-        photo_id=group_photo.pk,
554
-        msg_type=UserMessageInfo.THUMBUP,
555
-        msg_title=u'点赞',
556
-        msg_content=u'点赞',
557
-    )
642
+    # 判断群组照片发布者是否已经被管理员移除,如若移除,则不给发布者提醒
643
+    if r.sismember(GROUP_USERS_PASSED_SET % group_photo.group_id, group_photo.user_id):
644
+        UserMessageInfo.objects.create(
645
+            from_uid=user_id,
646
+            from_nickname=group_user.nickname,
647
+            from_avatar=group_user.avatar,
648
+            to_uid=group_photo.user_id,
649
+            group_id=group_photo.group_id,
650
+            photo_id=group_photo.pk,
651
+            msg_type=UserMessageInfo.THUMBUP,
652
+            msg_title=u'点赞',
653
+            msg_content=u'点赞',
654
+        )
558 655
 
656
+    # 群组照片点赞列表
559 657
     photo_thumbups = PhotoThumbUpInfo.objects.filter(
560 658
         photo_id=photo_id,
561 659
         thumbup=True,
@@ -572,33 +670,22 @@ def thumbup_submit_api(request):
572 670
 
573 671
 
574 672
 def thumbup_list_api(request):
673
+    """
674
+    飞图点赞列表
675
+    :param request:
676
+    :return:
677
+    """
575 678
     group_id = request.POST.get('group_id', '')
576 679
     user_id = request.POST.get('user_id', '')
577 680
     photo_id = request.POST.get('photo_id', '')
578 681
 
579
-    try:
580
-        group = GroupInfo.objects.get(group_id=group_id)
581
-    except GroupInfo.DoesNotExist:
582
-        return response(GroupStatusCode.GROUP_NOT_FOUND)
583
-
584
-    try:
585
-        group_user = GroupUserInfo.objects.get(group_id=group_id, user_id=user_id, user_status=GroupUserInfo.PASSED)
586
-    except GroupUserInfo.DoesNotExist:
587
-        return response(GroupStatusCode.GROUP_USER_NOT_FOUND)
588
-
589
-    try:
590
-        group_photo = GroupPhotoInfo.objects.get(pk=photo_id)
591
-    except GroupPhotoInfo.DoesNotExist:
592
-        return response(GroupPhotoStatusCode.MESSAGE_NOT_FOUND)
593
-
594
-    try:
595
-        thumbup = PhotoThumbUpInfo.objects.get(
596
-            photo_id=photo_id,
597
-            user_id=user_id,
598
-        ).thumbup
599
-    except PhotoThumbUpInfo.DoesNotExist:
600
-        thumbup = False
682
+    # user_id 是否点赞 photo_id
683
+    thumbup = PhotoThumbUpInfo.objects.filter(
684
+        photo_id=photo_id,
685
+        user_id=user_id,
686
+    ).exists()
601 687
 
688
+    # 群组照片点赞列表
602 689
     photo_thumbups = PhotoThumbUpInfo.objects.filter(
603 690
         photo_id=photo_id,
604 691
         thumbup=True,
@@ -615,25 +702,28 @@ def thumbup_list_api(request):
615 702
 
616 703
 
617 704
 def thumbup_cancel_api(request):
705
+    """
706
+    飞图点赞取消
707
+    :param request:
708
+    :return:
709
+    """
618 710
     group_id = request.POST.get('group_id', '')
619 711
     user_id = request.POST.get('user_id', '')
620 712
     photo_id = request.POST.get('photo_id', '')
621 713
 
622
-    try:
623
-        group = GroupInfo.objects.get(group_id=group_id)
624
-    except GroupInfo.DoesNotExist:
625
-        return response(GroupStatusCode.GROUP_NOT_FOUND)
626
-
714
+    # 群组用户校验
627 715
     try:
628 716
         group_user = GroupUserInfo.objects.get(group_id=group_id, user_id=user_id, user_status=GroupUserInfo.PASSED)
629 717
     except GroupUserInfo.DoesNotExist:
630
-        return response(GroupStatusCode.GROUP_USER_NOT_FOUND)
718
+        return response(GroupUserStatusCode.GROUP_USER_NOT_FOUND)
631 719
 
720
+    # 群组照片校验
632 721
     try:
633 722
         group_photo = GroupPhotoInfo.objects.get(pk=photo_id)
634 723
     except GroupPhotoInfo.DoesNotExist:
635
-        return response(GroupPhotoStatusCode.MESSAGE_NOT_FOUND)
724
+        return response(GroupPhotoStatusCode.GROUP_PHOTO_NOT_FOUND)
636 725
 
726
+    # 群组照片点赞取消
637 727
     photo_thumbup, created = PhotoThumbUpInfo.objects.get_or_create(
638 728
         photo_id=photo_id,
639 729
         user_id=user_id,
@@ -641,21 +731,25 @@ def thumbup_cancel_api(request):
641 731
     photo_thumbup.thumbup = False
642 732
     photo_thumbup.save()
643 733
 
734
+    # 群组照片点赞数更新
644 735
     group_photo.thumbup_num -= 1
645 736
     group_photo.save()
646 737
 
647
-    UserMessageInfo.objects.create(
648
-        from_uid=user_id,
649
-        from_nickname=group_user.nickname,
650
-        from_avatar=group_user.avatar,
651
-        to_uid=group_photo.user_id,
652
-        group_id=group_photo.group_id,
653
-        photo_id=group_photo.pk,
654
-        msg_type=UserMessageInfo.THUMBUP,
655
-        msg_title=u'取消点赞',
656
-        msg_content=u'取消点赞',
657
-    )
738
+    # 判断群组照片发布者是否已经被管理员移除,如若移除,则不给发布者提醒
739
+    if r.sismember(GROUP_USERS_PASSED_SET % group_photo.group_id, group_photo.user_id):
740
+        UserMessageInfo.objects.create(
741
+            from_uid=user_id,
742
+            from_nickname=group_user.nickname,
743
+            from_avatar=group_user.avatar,
744
+            to_uid=group_photo.user_id,
745
+            group_id=group_photo.group_id,
746
+            photo_id=group_photo.pk,
747
+            msg_type=UserMessageInfo.THUMBUP,
748
+            msg_title=u'取消点赞',
749
+            msg_content=u'取消点赞',
750
+        )
658 751
 
752
+    # 群组照片点赞列表
659 753
     photo_thumbups = PhotoThumbUpInfo.objects.filter(
660 754
         photo_id=photo_id,
661 755
         thumbup=True,
@@ -672,10 +766,16 @@ def thumbup_cancel_api(request):
672 766
 
673 767
 
674 768
 def pai2_home_api(request):
769
+    """
770
+    首页信息
771
+    :param request:
772
+    :return:
773
+    """
675 774
     user_id = request.POST.get('user_id', '')
676 775
     page = int(request.POST.get('page', 1))
677 776
     num = int(request.POST.get('num', settings.PAI2_HOME_PER_PAGE))
678 777
 
778
+    # 执行原生 SQL 语句,获取首页照片列表
679 779
     cursor = connection.cursor()
680 780
     cursor.execute(PAI2_HOME_API.format(
681 781
         user_id=user_id,
@@ -684,8 +784,10 @@ def pai2_home_api(request):
684 784
     ))
685 785
     rows = cursor.fetchall()
686 786
 
787
+    # 首页照片分页
687 788
     rows, left = pagination(rows, page, num)
688 789
 
790
+    # 首页照片信息
689 791
     rows = [{
690 792
         'group_id': row[0],
691 793
         'group_name': row[1],

+ 16 - 0
message/views.py

@@ -12,6 +12,11 @@ from utils.error.response_utils import response
12 12
 
13 13
 
14 14
 def message_list_api(request):
15
+    """
16
+    消息列表
17
+    :param request:
18
+    :return:
19
+    """
15 20
     user_id = request.POST.get('user_id', '') or request.GET.get('user_id', '')
16 21
 
17 22
     messages = UserMessageInfo.MESSAGE_TYPE_INFO
@@ -38,6 +43,12 @@ def message_list_api(request):
38 43
 
39 44
 
40 45
 def message_type_list_api(request, msg_type):
46
+    """
47
+    分类消息列表
48
+    :param request:
49
+    :param msg_type:
50
+    :return:
51
+    """
41 52
     user_id = request.POST.get('user_id', '') or request.GET.get('user_id', '')
42 53
     page = int(request.POST.get('page', 0) or request.GET.get('page', 0) or 1)
43 54
     num = int(request.POST.get('num', 0) or request.GET.get('num', 0) or settings.MESSAGE_NUM_PER_PAGE)
@@ -62,6 +73,11 @@ def message_type_list_api(request, msg_type):
62 73
 
63 74
 
64 75
 def message_read_api(request):
76
+    """
77
+    消息读取
78
+    :param request:
79
+    :return:
80
+    """
65 81
     pk = int(request.POST.get('pk', 0) or request.GET.get('pk', 0) or -1)
66 82
     user_id = request.POST.get('user_id', '') or request.GET.get('user_id', '')
67 83
 

+ 10 - 0
operation/views.py

@@ -6,6 +6,11 @@ from operation.models import LatestAppInfo, SplashInfo
6 6
 
7 7
 
8 8
 def upgrade_api(request):
9
+    """
10
+    APP 升级
11
+    :param request:
12
+    :return:
13
+    """
9 14
     try:
10 15
         appinfo = LatestAppInfo.objects.all()[0].data
11 16
     except IndexError:
@@ -21,6 +26,11 @@ def upgrade_api(request):
21 26
 
22 27
 
23 28
 def splash_api(request):
29
+    """
30
+    启动页面
31
+    :param request:
32
+    :return:
33
+    """
24 34
     splashes = SplashInfo.objects.all()
25 35
     splashes = [splash.data for splash in splashes]
26 36
 

+ 10 - 0
pay/views.py

@@ -22,6 +22,11 @@ wxpay = WeChatPay(WECHAT['appID'], WECHAT['apiKey'], WECHAT['mchID'])
22 22
 
23 23
 @transaction.atomic
24 24
 def wx_order_create_api(request):
25
+    """
26
+    订单创建
27
+    :param request:
28
+    :return:
29
+    """
25 30
     from_uid = request.POST.get('from_uid', '')
26 31
     to_lid = request.POST.get('to_lid', '')
27 32
     to_uid = request.POST.get('to_uid', '')
@@ -71,6 +76,11 @@ def order_paid_success(order):
71 76
 
72 77
 @transaction.atomic
73 78
 def wx_notify_url_api(request):
79
+    """
80
+    支付异步通知回调地址
81
+    :param request:
82
+    :return:
83
+    """
74 84
     try:
75 85
         data = xmltodict.parse(request.body)['xml']
76 86
     except xmltodict.ParsingInterrupted:

+ 30 - 4
photo/views.py

@@ -19,6 +19,9 @@ from utils.watermark_utils import watermark_wrap
19 19
 from utils.error.errno_utils import PhotoStatusCode
20 20
 from utils.error.response_utils import response
21 21
 
22
+from utils.redis.rkeys import GROUP_LAST_PHOTO_PK
23
+from utils.redis.rgroup import get_group_info, get_group_users_info
24
+
22 25
 from curtail_uuid import CurtailUUID
23 26
 
24 27
 from TimeConvert import TimeConvert as tc
@@ -27,7 +30,15 @@ import os
27 30
 import shortuuid
28 31
 
29 32
 
33
+r = settings.REDIS_CACHE
34
+
35
+
30 36
 def uuid_init(request):
37
+    """
38
+    生成唯一标识
39
+    :param request:
40
+    :return:
41
+    """
31 42
     num = int(request.GET.get('num', 1000))
32 43
 
33 44
     for i in xrange(num):
@@ -43,6 +54,11 @@ def uuid_init(request):
43 54
 # curl -X POST -F user=xxxxxxx -F num=100 http://api.pai.ai/uuid
44 55
 @transaction.atomic
45 56
 def uuid(request):
57
+    """
58
+    获取唯一标识
59
+    :param request:
60
+    :return:
61
+    """
46 62
     lensman_id = request.POST.get('user', '')
47 63
     num = int(request.POST.get('num', 100))
48 64
 
@@ -72,6 +88,11 @@ def uuid(request):
72 88
 #
73 89
 # curl -X POST -F user=xxxxxxx -F session=xxxxxxx -F photo_id=xxxxxxx -F photo=@xxxxxxx.jpg http://api.pai.ai/photos/upload
74 90
 def upload_photo(request):
91
+    """
92
+    上传图片
93
+    :param request:
94
+    :return:
95
+    """
75 96
     lensman_id = request.POST.get('user', '')
76 97
     session_id = request.POST.get('session', '')
77 98
     photo_id = request.POST.get('photo_id', '')
@@ -121,6 +142,12 @@ def upload_photo(request):
121 142
 
122 143
 
123 144
 def session_detail_api(request, session):
145
+    """
146
+    Session 详情
147
+    :param request:
148
+    :param session:
149
+    :return:
150
+    """
124 151
     user_id = request.POST.get('user_id', '')
125 152
 
126 153
     # 判断 user_id 是否存在,如果不存在,则直接分配帐户
@@ -142,9 +169,8 @@ def session_detail_api(request, session):
142 169
     group_id = group.group_id
143 170
 
144 171
     # 判断 group_id/user_id 的群组用户是否存在,如果不存在,则直接创建
145
-    group_photo = GroupPhotoInfo.objects.filter(group_id=group_id).last()
146 172
     group_user = GroupUserInfo.objects.get_or_create(group_id=group_id, user_id=user_id, defaults={
147
-        'current_id': group_photo and group_photo.id or -1,
173
+        'current_id': int(r.get(GROUP_LAST_PHOTO_PK % group_id) or -1),
148 174
         'nickname': user.final_nickname,
149 175
         'admin': False,
150 176
         'user_status': GroupUserInfo.PASSED,
@@ -159,9 +185,9 @@ def session_detail_api(request, session):
159 185
             'count': photos.count(),
160 186
             'photos': [p.detail for p in photos],
161 187
             'group_id': group_id,
162
-            'group': group.data,
188
+            'group': get_group_info(group_id),
163 189
             'user_id': user_id,
164
-            'users': group.users(user_id),
190
+            'users': get_group_users_info(group_id, user_id),
165 191
         }
166 192
     })
167 193
 

+ 0 - 1
utils/error/errno_utils.py

@@ -45,7 +45,6 @@ class GroupStatusCode(BaseStatusCode):
45 45
     NO_REFUSE_PERMISSION = StatusCodeField(402009, u'No Refuse Permission', description=u'没有拒绝权限')
46 46
     DUPLICATE_JOIN_REQUEST = StatusCodeField(402010, u'Duplicate Join Request', description=u'重复加群申请')
47 47
     JOIN_REQUEST_NOT_FOUND = StatusCodeField(402011, u'Join Request Not Found', description=u'加群申请不存在')
48
-    GROUP_USER_NOT_FOUND = StatusCodeField(402012, u'Group User Not Found', description=u'该用户不在群组')
49 48
 
50 49
 
51 50
 class GroupUserStatusCode(BaseStatusCode):

+ 2 - 1
utils/error/response_utils.py

@@ -6,6 +6,7 @@ from django.http import JsonResponse
6 6
 def response(status_code, data={}):
7 7
     return JsonResponse({
8 8
         'status': status_code,
9
-        'message': status_code.description,
9
+        'message': status_code.message,
10
+        'description': status_code.description,
10 11
         'data': data,
11 12
     })

+ 0 - 7
utils/redis/keys.py

@@ -1,7 +0,0 @@
1
-# -*- coding: utf-8 -*-
2
-
3
-# 用户相关
4
-PROFILE_USER_INFO = 'profile:user:info:%s'  # 用户信息,uid
5
-
6
-# 群组相关
7
-GROUP_INFO = 'group:info:%s'  # 群组信息,group_id

+ 33 - 0
utils/redis/retrieve.py

@@ -0,0 +1,33 @@
1
+# -*- coding: utf-8 -*-
2
+
3
+from django.conf import settings
4
+
5
+from group.models import GroupInfo, GroupUserInfo, GroupPhotoInfo
6
+
7
+from utils.redis.rkeys import GROUP_USERS_APPLYING_SET, GROUP_USERS_PASSED_SET, GROUP_USERS_REFUSED_SET, GROUP_USERS_DELETED_SET
8
+from utils.redis.rkeys import GROUP_LAST_PHOTO_PK
9
+
10
+
11
+r = settings.REDIS_CACHE
12
+
13
+
14
+def retrieve_redis_data():
15
+    # 群组/群组照片相关
16
+    groups = GroupInfo.objects.filter(status=True)
17
+    for group in groups:
18
+        group_photo = GroupPhotoInfo.objects.filter(group_id=group.group_id).last()
19
+        # 设置群组最后一张照片PK
20
+        r.set(GROUP_LAST_PHOTO_PK % group.group_id, group_photo and group_photo.pk or -1)
21
+
22
+    # 群组用户相关
23
+    group_users = GroupUserInfo.objects.filter(status=True)
24
+    for group_user in group_users:
25
+        group_id, user_id = group_user.group_id, group_user.user_id
26
+        if group_user.user_status == GroupUserInfo.PASSED:
27
+            r.sadd(GROUP_USERS_PASSED_SET % group_id, user_id)
28
+        elif group_user.user_status == GroupUserInfo.DELETED:
29
+            r.sadd(GROUP_USERS_DELETED_SET % group_id, user_id)
30
+        elif group_user.user_status == GroupUserInfo.APPLYING:
31
+            r.sadd(GROUP_USERS_APPLYING_SET % group_id, user_id)
32
+        elif group_user.user_status == GroupUserInfo.REFUSED:
33
+            r.sadd(GROUP_USERS_REFUSED_SET % group_id, user_id)

+ 68 - 0
utils/redis/rgroup.py

@@ -0,0 +1,68 @@
1
+# -*- coding: utf-8 -*-
2
+
3
+from django.conf import settings
4
+from django.core.serializers.json import DjangoJSONEncoder
5
+
6
+from utils.redis.rkeys import GROUP_INFO, GROUP_USERS_INFO, GROUP_LAST_PHOTO_PK
7
+
8
+import json
9
+
10
+
11
+r = settings.REDIS_CACHE
12
+
13
+
14
+# 群组信息相关
15
+
16
+
17
+def set_group_info(group):
18
+    """ 设置群组信息 """
19
+    group_data = group.data
20
+    r.set(GROUP_INFO % group.group_id, json.dumps(group_data, cls=DjangoJSONEncoder))
21
+    return group_data
22
+
23
+
24
+def set_group_info_by_id(group_id):
25
+    """ 设置群组信息 """
26
+    from group.models import GroupInfo
27
+    try:
28
+        group = GroupInfo.objects.get(group_id=group_id)
29
+    except GroupInfo.DoesNotExist:
30
+        return {}
31
+    return set_group_info(group)
32
+
33
+
34
+def get_group_info(group_id):
35
+    """ 获取群组信息 """
36
+    return json.loads(r.get(GROUP_INFO % group_id) or '{}') or set_group_info_by_id(group_id)
37
+
38
+
39
+# 群组用户信息相关
40
+
41
+
42
+def set_group_users_info(group):
43
+    """ 设置群组用户信息 """
44
+    group_users = group.users(admin=False)
45
+    r.set(GROUP_USERS_INFO % group.group_id, json.dumps({
46
+        'group_users': group_users,
47
+        'admin_id': group.admin_id,
48
+    }, cls=DjangoJSONEncoder))
49
+    return group_users
50
+
51
+
52
+def set_group_users_info_by_id(group_id):
53
+    """ 设置群组用户信息 """
54
+    from group.models import GroupInfo
55
+    try:
56
+        group = GroupInfo.objects.get(group_id=group_id)
57
+    except GroupInfo.DoesNotExist:
58
+        return {}
59
+    return set_group_users_info(group)
60
+
61
+
62
+def get_group_users_info(group_id, user_id):
63
+    """ 获取群组用户信息 """
64
+    group_users_data = json.loads(r.get(GROUP_USERS_INFO % group_id) or '{}') or set_group_users_info_by_id(group_id)
65
+    group_users, admin_id = group_users_data.get('group_users', {}), group_users_data.get('admin_id', '')
66
+    if group_users and user_id != admin_id:
67
+        [group_users.pop(k) for k in ['applying_count', 'applying']]
68
+    return group_users

+ 17 - 0
utils/redis/rkeys.py

@@ -0,0 +1,17 @@
1
+# -*- coding: utf-8 -*-
2
+
3
+# 用户相关
4
+PROFILE_INFO = 'profile:info:%s'  # STRING,用户信息,uid
5
+
6
+# 群组相关
7
+GROUP_INFO = 'group:info:%s'  # STRING,群组信息,group_id
8
+
9
+# 群组用户相关
10
+GROUP_USERS_INFO = 'group:users:info:%s'  # STRING,群组用户信息,group_id
11
+GROUP_USERS_APPLYING_SET = 'group:users:applying:set:%s'  # SET,群组用户申请集合,group_id
12
+GROUP_USERS_PASSED_SET = 'group:users:passed:set:%s'  # SET,群组用户通过集合,group_id
13
+GROUP_USERS_REFUSED_SET = 'group:users:refused:set:%s'  # SET,群组用户拒绝集合,group_id
14
+GROUP_USERS_DELETED_SET = 'group:users:deleted:set:%s'  # SET,群组用户移除集合,group_id
15
+
16
+# 群组照片相关
17
+GROUP_LAST_PHOTO_PK = 'group:last:photo:pk:%s'  # STRING,群组最后一张照片PK,group_id